# -*- coding:utf-8 -*-

# doc2pdf.py: python script to convert doc to pdf with bookmarks!
# Requires python for win32 extension

import sys, os
from win32com.client import Dispatch, constants, gencache


def doc2pdf(input, output):
    w = Dispatch("Word.Application")

    try:
        doc = w.Documents.Open(input, ReadOnly=1)
        doc.ExportAsFixedFormat(output, constants.wdExportFormatPDF,Item=constants.wdExportDocumentWithMarkup,CreateBookmarks=constants.wdExportCreateHeadingBookmarks)
        return 0
    except:
        return 1
    finally:
        w.Quit(constants.wdDoNotSaveChanges)

    # Generate all the support we can.


def GenerateSupport():
    # enable python COM support for Word 2007
    # this is generated by: makepy.py -i "Microsoft Word 12.0 Object Library"
    gencache.EnsureModule('{00020905-0000-0000-C000-000000000046}', 0, 8, 4)


def main():
    #如果参数的数量为2（.py 所运行的程序文件为第一个参数，之后的参数为所带的参数）
    #参数为2的话则进输出采用默认的名字进行保存
    if (len(sys.argv) == 2):
        # print('第一个参数为：' + sys.argv[0])
        # print('第二个参数为：' + sys.argv[1])

        input = sys.argv[1]
        output = os.path.splitext(input)[0] + '.pdf'
    #多加参数进行输出文件名字的自定义
    elif (len(sys.argv) == 3):
        input = sys.argv[1]
        output = sys.argv[2]
    else:
        input = u'BA06007013.docx'  # word文档的名称
        output = u'BA06007013.pdf'  # pdf文档的名称
    if (not os.path.isabs(input)):
        input = os.path.abspath(input)
    if (not os.path.isabs(output)):
        output = os.path.abspath(output)
    try:
        GenerateSupport()
        rc = doc2pdf(input, output)
        return rc
    except:
        return -1


if __name__ == '__main__':
    rc = main()
    if rc:
        sys.exit(rc)
    sys.exit(0)